package SubjectTree.One;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;

import Utility.Node;

public class LevelOrder2 {

/**
 * 难度：中等
 * 
 * 429. N 叉树的层序遍历
 * 	给定一个 N 叉树，返回其节点值的层序遍历。（即从左到右，逐层遍历）。
 * 	树的序列化输入是用层序遍历，每组子节点都由 null 值分隔（参见示例）。
 * 	
 * 示例 1：
 * 	输入：root = [1,null,3,2,4,null,5,6]
 * 	输出：[[1],[3,2,4],[5,6]]
 * 	
 * 示例 2：
 * 	输入：root = [1,null,2,3,4,5,null,null,6,7,null,8,null,9,10,null,null,11,null,12,null,13,null,null,14]
 * 	输出：[[1],[2,3,4,5],[6,7,8,9,10],[11,12,13],[14]]
 * 	
 * 提示：
 * 	树的高度不会超过 1000
 * 	树的节点总数在 [0, 10^4] 之间
 *
 * */
	
	public static void main(String[] args) {
		// TODO Auto-generated method stub
		List<Node> list = new ArrayList<>();
		List<Node> list2 = new ArrayList<>();
		LevelOrder2 lo = new LevelOrder2();
		Node node = new Node(1);
		node.children = list;
		Node node3 = new Node(3);
		node3.children = list2;
		list.add(new Node(2));
		list.add(new Node(4));
		list.add(node3);
		list2.add(new Node(5));
		list2.add(new Node(6));
		System.out.println(lo.levelOrder(node));
	}

	public List<List<Integer>> levelOrder(Node root) {
		List<List<Integer>> ret = new ArrayList<>();
		if(root==null)return ret;
		Queue<Node> queue = new LinkedList<>();
		queue.offer(root);
		while(!queue.isEmpty()) {
			List<Integer> list = new ArrayList<>();
			int currentLevelSize = queue.size();
			for(int i=0;i<currentLevelSize;i++) {
				Node node = queue.poll();
				list.add(node.val);
				if(node.children!=null) {
					node.children.stream().forEach(n->queue.offer(n));
				}
			}
			ret.add(list);
		}
		return ret;
    }
}
